import pymysql

# coding:utf-8
import pandas as pd
import csv
import re
import numpy as np
from datetime import datetime

dou1 = pd.read_excel(r'G:\工作\2025年订单\10月\财务匹配\2025年10月抖音1店资金账单.xlsx', dtype={'子订单号': str})
dou1['平台'] = '抖音1'

dou2 = pd.read_excel(r'G:\工作\2025年订单\10月\财务匹配\2025年10月抖音2店资金账单.xlsx', dtype={'子订单号': str})
dou2['平台'] = '抖音2'

dou3 = pd.read_excel(r'G:\工作\2025年订单\10月\财务匹配\2025年10月抖音3店资金账单.xlsx', dtype={'子订单号': str})
dou3['平台'] = '抖音3'

dou4 = pd.read_excel(r'G:\工作\2025年订单\10月\财务匹配\2025年10月抖音4店资金账单.xlsx', dtype={'子订单号': str})
dou4['平台'] = '抖音4'

dou = pd.concat([dou1, dou2, dou3, dou4])

# 读取文件
# 打开数据库连接
db = pymysql.connect(host="localhost", user="root", password="hui123456", db='dbtest')
# 使用cursor()方法创建一个游标对象cursor
cursor = db.cursor()

sql = """
    SELECT 
            orderid2, 
            zqudao
    FROM orderid_qd
"""


cursor.execute(sql)
result = cursor.fetchall()
qd = pd.DataFrame(list(result))
qd.columns = ['子订单号', '最终渠道']

res = pd.merge(dou, qd, on='子订单号', how='left')

for i in range(1, 5):
    res_dou = res[res['平台'] == f'抖音{i}']

    res_dou.to_excel(r'G:\工作\2025年订单\10月\财务匹配\渠道区分\2025年10月抖音{}店资金帐单匹配结果.xlsx'.format(i))

# 抖音结算账单渠道拆分——财务

# 注意：
# 财务提供结算帐单，对订单中订单id进行处理，
# 子订单号缺失的如果主订单中有，填充主订单号，无则空白。
# 最后将id号格式调整，匹配数据库（orderid_qd）中的订单渠道